#include "eye_n_d.h"
using namespace Eye;

extern "C" __global__ __aicore__ void eye(GM_ADDR y, GM_ADDR workspace, GM_ADDR tiling) {
    if (TILING_KEY_IS(101)) {
        GET_TILING_DATA(tilingData, tiling);
        EyeND<DTYPE_Y> op;
        op.Init(y, &tilingData);
        op.ProcessSmallM();
    } else if (TILING_KEY_IS(201)) {
        GET_TILING_DATA(tilingData, tiling);
        EyeND<DTYPE_Y> op;
        op.Init(y, &tilingData);
        op.ProcessMidM();
    }
}